import numpy as np
from CR_DSPPytorch import cal_levels_for_APoT

if __name__ == '__main__':
    b = 4
    k = 2
    levels = cal_levels_for_APoT(b, k)
    print(np.sort(levels))
    print(np.linspace(0, 1, 2**b))

    # for indx in range(b//k):
    #     print(1/p[indx, :])